import os
import shutil

# 定义多个原始图片目录的列表，你可以在这里添加更多的目录路径
source_dirs = ["data_pet_zhu1", "data_pet_zhu2"]
# 定义新的目标目录
target_dir = "data_pet_zhu"

# 如果目标目录不存在，则创建它
if not os.path.exists(target_dir):
    os.makedirs(target_dir)

# 用于记录新文件名的序号
count = 1

# 遍历每个原始图片目录
for source_dir in source_dirs:
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            if file.endswith('.jpg'):
                # 构建源文件的完整路径
                source_file_path = os.path.join(root, file)
                # 构建新的文件名，格式为序号.jpg，例如：1.jpg、2.jpg等
                new_file_name = f"{count}.jpg"
                # 构建目标文件的完整路径
                target_file_path = os.path.join(target_dir, new_file_name)
                # 复制文件到目标目录
                shutil.copy2(source_file_path, target_file_path)
                count += 1

print(f"已成功将多个目录中的jpg图片重命名并复制到新目录 {target_dir} 中。")